neural network cost function python code